3d function in python